//
//  UIImage+Extension.swift
//  醒浪微博
//
//  Created by 雷顺哲 on 16/7/17.
//  Copyright © 2016年 LSZ. All rights reserved.
//

import UIKit

extension UIImage {
    /**
    //  截取屏幕快照
    
    - returns: 返回截取image
    */
    
    class func getScreenShot() -> UIImage {
        
//        获取主window
        let window = UIApplication.sharedApplication().keyWindow!
        
//        开启图片上下文
        
        UIGraphicsBeginImageContext(window.size)
        
//        把window的内容渲染到图片上下文中
        window.drawViewHierarchyInRect(window.bounds, afterScreenUpdates: false)
        
//        从上下文获取图片
        let image = UIGraphicsGetImageFromCurrentImageContext()
        
        
//        关闭图片上下文
        UIGraphicsEndImageContext()
        return image
        
    }
    
    
    /**
     根据指定压缩后的宽度,压缩图片
     
     - parameter width: 压缩的宽度
     - parameter image: 需要压缩的图片
     
     - returns:   压缩后的图片
     */
     func scaleImageWithScaleWidth(scaleWidth: CGFloat) -> UIImage {
        
        //        根据压缩后的宽度计算压缩后的高度
        let scaleHeight = scaleWidth / self.size.width * self.size.height
        
        //        压缩后的大小
        let size = CGSize(width: scaleWidth, height: scaleHeight)
        
        //        开启上下文
        UIGraphicsBeginImageContext(size)
        
        //        把图片渲染到指定范围中
        self.drawInRect(CGRect(origin: CGPointZero, size: size))
        //        后去上下文压缩后的图片
        let scaleImage = UIGraphicsGetImageFromCurrentImageContext()
        //        关闭上下文
        
        UIGraphicsEndImageContext()
        
        return scaleImage
    }
    
    
}